package com.test.ai.config;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.test.ai.dao.RedisChatMemory;
import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
import org.springframework.ai.chat.memory.ChatMemory;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.serializer.GenericJackson2JsonRedisSerializer;
import org.springframework.data.redis.serializer.StringRedisSerializer;

/**
 * Created with IntelliJ IDEA.
 *
 * @description:
 * @author: liuziyang
 * @since: 8/21/25 4:59 PM
 * @modifiedBy:
 * @version: 1.0
 */
@Configuration
public class RedisChatMemoryConfig {
  @Bean
  public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory redisConnectionFactor) {
    RedisTemplate<String, Object> redisTemplate = new RedisTemplate<>();
    redisTemplate.setConnectionFactory(redisConnectionFactor);
    // 设置key序列化方式string
    redisTemplate.setKeySerializer(new StringRedisSerializer());
    // 设置value的序列化方式json，使用GenericJackson2JsonRedisSerializer替换默认序列化
    redisTemplate.setValueSerializer(new GenericJackson2JsonRedisSerializer());
    redisTemplate.setHashKeySerializer(new StringRedisSerializer());
    redisTemplate.setHashValueSerializer(new GenericJackson2JsonRedisSerializer());
    redisTemplate.afterPropertiesSet();
    return redisTemplate;
  }

  /** 注册 RedisChatMemory Bean */
  @Bean("inRedis")
  public RedisChatMemory redisChatMemory(
      RedisTemplate<String, String> redisTemplate, ObjectMapper objectMapper) {
    return new RedisChatMemory(redisTemplate, objectMapper);
  }

  /**
   * Message chat memory advisor message chat memory advisor.
   *
   * @return the message chat memory advisor
   */
  @Bean("inRedisAdvisor")
  public MessageChatMemoryAdvisor messageChatMemoryAdvisor(
      @Qualifier("inRedis") ChatMemory chatMemory) {
    return new MessageChatMemoryAdvisor(chatMemory);
  }
}
